package cn.tedu.boot42.controller;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

@RestController
public class UploadController {

    @RequestMapping("upload")
    public String upload(MultipartFile picFile) throws IOException {
        System.out.println("picFile = "+picFile);
        //得到文件原始文件名 a.jpg
        String filename = picFile.getOriginalFilename();
        //得到后缀名,从最后一个.开始截取
        String suffix = filename.substring(filename.lastIndexOf("."));
        //得到唯一的文件名,UUID.randomUUID()得到一个唯一标识符
        filename = UUID.randomUUID() + suffix;
        System.out.println("fileName:"+filename);
        //准备保存图片的文件夹路径
        String dirPath = "D:/files";
        File dirFile = new File(dirPath);
        //如果文件夹不存在,则创建新的文件夹
        if (!dirFile.exists()) {
            dirFile.mkdirs(); //创建文件夹
        }
        //得到文件的完整路径
        String filePath = dirPath + "/" + filename;
        //把文件保存到此路径
        picFile.transferTo(new File(filePath));
        System.out.println("文件保存成功!请在此路径中查看 "+filePath);
        //返回文件名,因为删除会用到
        return filename;
    }

    @RequestMapping("remove")
    public void remove(String name) {
        String filePath = "D:/files/"+name;
        new File(filePath).delete(); //删除文件
    }
}
